package com.qf.commons.web.exception;

import com.qf.commons.web.utils.RequestUtils;
import com.qf.data.r.Codes;
import com.qf.data.r.R;
import com.qf.data.r.RUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.BindException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import javax.servlet.http.HttpServletRequest;
import javax.validation.ConstraintViolationException;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * description: RestController层的异常统一处理
 * author: Ken
 * 公众号：Java架构栈
 */
@RestControllerAdvice
@Slf4j
public class GlobalException {

    /**
     * 参数校验异常的统一处理
     * @return
     */
    @ExceptionHandler(BindException.class)
    public R bindExceptionHandler(BindException exception){
        //获取参数校验的异常信息，并且返回给客户端
        Set<String> errors = exception.getBindingResult().getAllErrors()
                .stream()
                .map(objectError -> objectError.getDefaultMessage())
                .collect(Collectors.toSet());
        //返回错误信息
        return RUtils.create(Codes.PARAMENT_ERROR, errors);
    }

    /**
     * 基于@RequestBody接收参数时的参数校验异常
     * @return
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public R methodArgumentNotValidExceptionHandler(MethodArgumentNotValidException exception){
        //获取参数校验的异常信息，并且返回给客户端
        Set<String> errors = exception.getBindingResult().getAllErrors()
                .stream()
                .map(objectError -> objectError.getDefaultMessage())
                .collect(Collectors.toSet());
        //返回错误信息
        return RUtils.create(Codes.PARAMENT_ERROR, errors);
    }

    /**
     * 基于形参列表进行校验的异常处理
     * @return
     */
    @ExceptionHandler(ConstraintViolationException.class)
    public R constraintViolationExceptionHandler(ConstraintViolationException exception){
        Set<String> errors = exception.getConstraintViolations().stream()
                .map(constraintViolation -> constraintViolation.getMessage())
                .collect(Collectors.toSet());
        //返回错误信息
        return RUtils.create(Codes.PARAMENT_ERROR, errors);
    }

    /**
     * 统一异常处理方法
     * @return
     */
    @ExceptionHandler(Throwable.class)
    public R exceptionHandler(Throwable t){
        try {
            //获取当前的请求的url
            HttpServletRequest request = RequestUtils.getHttpServletRequest();

            //获取请求的url
            if (request != null) {
                String uri = request.getRequestURL().toString();
                log.error("[Global-Exception] - 捕获到全局异常信息，url：" + uri, t);
            }
        } catch (Exception e) {
            log.error("[Global-Exception] - 捕获到全局异常！", t);
        }

        //统一返回结果
        return RUtils.create(Codes.FAIL);
    }
}
